package com.easy.util;

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;

public class ExpiryMap {
    //用来存储数据的容器
    private Map<String, Object> map = new HashMap<String, Object>();
    //存储数据的有效期
    private Map<String, Long> expiryMap = new HashMap<String, Long>();
    //检查过期线程(池)
    private ScheduledExecutorService service = Executors.newSingleThreadScheduledExecutor();
    //检查周期
    private long expiryTime = 1000 * 6 ;

    //构造方法，启动定时任务，每隔expiryTime执行一次clearExpiry()方法
    //固定周期性执行任务，当任务的执行时长大于周期，那么下一个周期任务将在上一个执行完毕之后马上执行。
    public ExpiryMap() {
        service.scheduleAtFixedRate(this::clearExpiry,expiryTime,expiryTime, TimeUnit.MILLISECONDS);
    }
    //带自定义时间的构造方法
    public ExpiryMap(long expiryTime) {
        this.expiryTime = expiryTime;
        service.scheduleAtFixedRate(this::clearExpiry,expiryTime,expiryTime, TimeUnit.MILLISECONDS);
    }

    // 使用默认过期时间存储键值对
    public void put(String key, Object value) {
        map.put(key, value);
        expiryMap.put(key, System.currentTimeMillis() + expiryTime);
    }

    // 使用自定义过期时间存储键值对
    public void put(String key, Object value,long expiryTime) {
        map.put(key, value);
        expiryMap.put(key, System.currentTimeMillis() + expiryTime);
    }


    public Object get(String key) {
        // 检查key是否已过期
        if (expiryMap.get(key) !=null&&expiryMap.get(key) < System.currentTimeMillis()) {
            // 已过期则移除该key
            map.remove(key);
            expiryMap.remove(key);
        }
        // 返回当前key对应的值（可能为null）
        return map.get(key);
    }
    //同时从map和expiryMap中移除指定 key 的数据，彻底删除该键值对
    public void remove(String key) {
        map.remove(key);
        expiryMap.remove(key);
    }
    public boolean containsKey(String key) {
        // 先检查是否过期
        if (expiryMap.get(key) !=null&&expiryMap.get(key) < System.currentTimeMillis()) {
            map.remove(key);
            expiryMap.remove(key);
        }
        // 返回是否包含该key
        return map.containsKey(key);
    }
    //清除掉过期的Key
    public void clearExpiry(){
        // 遍历所有key检查是否过期
        for (String key : expiryMap.keySet()) {
            if (expiryMap.get(key) !=null&&expiryMap.get(key) < System.currentTimeMillis()) {
                // 过期则移除
                map.remove(key);
                expiryMap.remove(key);
            }
        }
    }
}